Python code to Print an Identity Matrix